// Copyright (c) 2024, BlockProject 3D

//

// All rights reserved.

//

// Redistribution and use in source and binary forms, with or without modification,

// are permitted provided that the following conditions are met:

//

//     * Redistributions of source code must retain the above copyright notice,

//       this list of conditions and the following disclaimer.

//     * Redistributions in binary form must reproduce the above copyright notice,

//       this list of conditions and the following disclaimer in the documentation

//       and/or other materials provided with the distribution.

//     * Neither the name of BlockProject 3D nor the names of its contributors

//       may be used to endorse or promote products derived from this software

//       without specific prior written permission.

//

// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS

// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT

// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR

// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR

// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,

// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,

// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR

// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF

// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING

// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS

// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

use crate::compiler::message::Message;

use crate::compiler::Protocol;

use crate::gen::base::map::TypePathMapper;

use crate::gen::base::message::{gen_msg_field_decl, generate, Templates};

use crate::gen::base::Error;

use crate::gen::codec::CodecMap;

use crate::gen::swift::util::{SwiftTypeMapper, SwiftUtils};

use crate::gen::template::Template;

use itertools::Itertools;

const TEMPLATE: &[u8] = include_bytes!("decl.template");

const TEMPLATE_EXT: &[u8] = include_bytes!("ext.template");

fn gen_initializer(

    templates: &Templates,

    msg: &Message,

    type_path_map: &TypePathMapper<SwiftTypeMapper>,

) -> Result<String, Error> {

    let init_field_list = msg

        .fields

        .iter()

        .map(|field| gen_msg_field_decl::<SwiftUtils, _>(field, templates, type_path_map))

        .collect::<Result<Vec<String>, Error>>()?

        .into_iter()

        .map(|v| v[..v.len() - 1].to_string())

        .join(", ");

    let initializers = msg

        .fields

        .iter()

        .map(|field| templates.template.scope().var("name", &field.name).render("decl", &["initializer"]).unwrap())

        .join("");

    Ok(templates

        .template

        .scope()

        .var("init_field_list", init_field_list)

        .var("initializers", initializers)

        .render("", &["decl"])

        .unwrap())

}

pub fn gen_message_decl(proto: &Protocol, codec_map: &CodecMap, msg: &Message) -> Result<String, Error> {

    let type_path_map = TypePathMapper::new(&proto.type_path_map, SwiftTypeMapper::from_protocol(proto));

    let mut templates = Templates {

        template: Template::compile(TEMPLATE_EXT).unwrap(),

        codec_map,

    };

    templates.template.var("proto_name", proto.name()).var("msg_name", &msg.name);

    let initializer = gen_initializer(&templates, msg, &type_path_map)?;

    templates.template = Template::compile(TEMPLATE).unwrap();

    templates.template.var("proto_name", proto.name()).var("initializer", initializer);

    generate::<SwiftUtils, _>(templates, msg, &type_path_map)

}